package com.jt.interceptor;

import com.jt.pojo.User;
import com.jt.util.CookieUtil;
import com.jt.util.ObjectMapperUtil;
import com.jt.util.UserThreadLocal;
import org.jboss.netty.util.internal.StringUtil;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.stereotype.Component;
import org.springframework.util.StringUtils;
import org.springframework.web.servlet.HandlerInterceptor;
import redis.clients.jedis.Jedis;
import redis.clients.jedis.JedisCluster;

import javax.servlet.http.Cookie;
import javax.servlet.http.HttpServletRequest;
import javax.servlet.http.HttpServletResponse;

@Component
public class UserInterceptor implements HandlerInterceptor {
    private static final String JT_TICKET = "JT_TICKET";
    @Autowired
    private Jedis jedisCluster;
    /**
     * 配置拦截器
     * 返回值说明:
     *      false->表示拦截 一般都要配合重定向使用
     *      true->表示放行
     * 如何实现业务?
     * 判断用户是否登录: Cookie是否有数据 检查redis数据
     * 重定向到系统登录页面
     */
    @Override
    public boolean preHandle(HttpServletRequest request, HttpServletResponse response, Object handler) throws Exception {
        //1.校验Cookie中是否有结果
        Cookie cookie = CookieUtil.getCookie(request, JT_TICKET);
        //2.校验Cookie是否有效
        if (cookie != null) {
            String ticket = cookie.getValue();
            if (StringUtils.hasLength(ticket)){
                //执行后续任务->校验redis中是否有结果
                if (jedisCluster.exists(ticket)){
                    String json = jedisCluster.get(ticket);
                    User user = ObjectMapperUtil.toObject(json, User.class);
                    //方案一:---->request对象的方式传递User
                    request.setAttribute("JT_USER",user);

                    //方案二:---->本地线程变量的方式传递User
                    /**注意rpc远程调用时,会开辟新的线程,
                     * 此时无法使用threadLocal传递,必须使用request对象*/
                    UserThreadLocal.setUser(user);
                    //表示用户登录过,返回true
                    return true;
                }
            }
        }
        //没有结果,则表示Cookie数据有误,应该删除
        CookieUtil.addCookie(response,JT_TICKET,"","/","jt.com",0);
        //3.如果数据为空,则重定向到登录页面
        response.sendRedirect("/user/login.html");
        return false;
    }

    /**
     * 为了防止内存泄漏,将多余的数据删除
     */
    @Override
    public void afterCompletion(HttpServletRequest request, HttpServletResponse response, Object handler, Exception ex) throws Exception {
        request.removeAttribute("JT_USER");
        UserThreadLocal.remove();
    }
}
